quadratic equation in python